# 作者: 王道 龙哥
# 2024年04月10日11时26分25秒
# xxx@qq.com
#!/usr/bin/python
# author luke

import numpy as np
from matplotlib import pyplot as plt

us_file_path = "./youtube_video_data/US_video_data_numbers.csv"
uk_file_path = "./youtube_video_data/GB_video_data_numbers.csv"

# t1 = np.loadtxt(us_file_path,delimiter=",",dtype="int",unpack=True)
# unpack实现转置的效果
t_us = np.loadtxt(us_file_path, delimiter=",", dtype="int")

#取美国电影评论数据
t_us_comment = t_us[:, -1]

# 取出来美国的youtube评论数和喜欢数,第1列是like数，第3列是comment数
t_us = t_us[t_us[:, 1] <= 250000]  # 筛选like数小于等于250000的视频
print(t_us.shape)
#取美国电影评论数据
t_us_like = t_us[:, 1]

t_us_comment = t_us[:, -1]

plt.figure(figsize=(20, 8), dpi=80)
# 打印横轴纵轴标签
plt.xlabel("like")
plt.ylabel("comment")
plt.scatter(t_us_like, t_us_comment)  # 画散点图,观察横轴和纵轴的关系

plt.show()
